package rabbitmq.rpc;

import com.rabbitmq.client.*;
import rabbitmq.constant.Constants;

import java.io.IOException;
import java.util.concurrent.TimeoutException;

/**
 * RPC通信模式的服务端
 * 1、接收请求
 * 2、发送响应
 */
public class RPCServer {
    public static void main(String[] args) throws IOException, TimeoutException {
        // 1、建立连接
        ConnectionFactory connectionFactory = new ConnectionFactory();
        connectionFactory.setHost(Constants.HOST); // 设置IP
        connectionFactory.setPort(Constants.PORT); // 设置开放的端口号，需要提前开放
        connectionFactory.setUsername(Constants.USER_NAME); // 设置RabbitMQ 的账号
        connectionFactory.setPassword(Constants.PASSWORD); // 设置RabbitMQ 的密码
        connectionFactory.setVirtualHost(Constants.VIRTUAL_HOST); // 设置 虚拟主机
        Connection connection = connectionFactory.newConnection();
        // 2、开启信道，也就是根据 连接创建信道
        Channel channel = connection.createChannel();
        // 3、交换机使用的内置的交换机，不需要进行声明
        // 4、声明队列
        channel.queueDeclare(Constants.RPC_REQUEST_QUEUE,true,false,false,null);
        channel.queueDeclare(Constants.RPC_RESPONSE_QUEUE,true,false,false,null);
        // 5、接收请求
        // 设置同时最多只能获取⼀个消息
        channel.basicQos(1);
        DefaultConsumer consumer = new DefaultConsumer(channel) {
            @Override
            public void handleDelivery(String consumerTag, Envelope envelope,
                                       AMQP.BasicProperties properties, byte[] body) throws IOException {
                String request = new String(body);
                System.out.println("接收到请求:"+ request);
                String response = "针对request:"+ request +", 响应成功";
                AMQP.BasicProperties basicProperties = new AMQP.BasicProperties().builder()
                        .correlationId(properties.getCorrelationId())
                        .build();
                channel.basicPublish("",Constants.RPC_RESPONSE_QUEUE,basicProperties,response.getBytes());
                // 手动确认
                // 这里的 boolean类型 是表示是否批量处理
                channel.basicAck(envelope.getDeliveryTag(), false);
            }
        };
        // 这里不进行自动的确认，使用手动确认
        channel.basicConsume(Constants.RPC_REQUEST_QUEUE, false,consumer);
    }
}
